package com.itheima.service;

import com.alibaba.dubbo.config.annotation.Service;
import com.github.pagehelper.Page;
import com.github.pagehelper.PageHelper;
import com.itheima.dao.MenuDao;
import com.itheima.dao.PermissionDao;
import com.itheima.dao.RoleDao;
import com.itheima.dao.UserDao;
import com.itheima.entity.PageResult;
import com.itheima.exceptions.BusinessException;
import com.itheima.pojo.Menu;
import com.itheima.pojo.Permission;
import com.itheima.pojo.Role;
import com.itheima.pojo.User;
import org.springframework.transaction.annotation.Transactional;

import javax.annotation.Resource;
import java.util.LinkedHashSet;
import java.util.List;
import java.util.Set;

/**
 * @author W.J.Y
 */
@Service(version = "1.0.0",interfaceClass = RoleService.class)
@Transactional
public class RoleServiceImpl implements RoleService{

    @Resource
    private RoleDao roleDao;
    @Resource
    private MenuDao menuDao;
    @Resource
    private PermissionDao permissionDao;
    @Resource
    private UserDao userDao;

    /**
     * 查询全部角色信息
     * @return  全部角色信息
     */
    @Override
    public List<Role> findAll() {
        return roleDao.findAll();
    }

    /**
     * 添加角色
     * @param role  角色实例
     * @throws BusinessException    可能出现的业务异常
     */
    @Override
    public void add(Role role) throws BusinessException {
        Integer count = roleDao.findRoleByKeyWord(role);
        if(count>0){
            throw new BusinessException("已存在相同代码的角色，添加失败！");
        }
        //首先首先添加角色
        roleDao.add(role);
        //添加角色与菜单、权限的关联关系
        setRoleRelevance(role);
    }

    /**
     * 添加角色与菜单、权限的关联关系
     * @param role  角色实例
     */
    private void setRoleRelevance(Role role) {
        Integer id = role.getId();
        //再添加菜单与角色的关联
        LinkedHashSet<Integer> menuIds = role.getMenuIds();
        if (menuIds!=null){
            for (Integer menuId : menuIds) {
                roleDao.setMenuAndRole(id,menuId);
            }
        }
        //再添加权限与角色的关联
        Set<Integer> permissionIds = role.getPermissionIds();
        if (permissionIds!=null){
            for (Integer permissionId : permissionIds) {
                roleDao.setPermissionAndRole(id,permissionId);
            }
        }
    }

    /**
     * 编辑角色
     * @param role  角色实例
     * @throws BusinessException    可能出现的业务异常
     */
    @Override
    public void update(Role role)  throws BusinessException{
        Integer count = roleDao.findRoleByKeyWord(role);
        if(count>0){
            throw new BusinessException("已存在相同代码的角色，编辑失败！");
        }
        roleDao.update(role);
        //首先删除角色与菜单、权限的关联关系
        roleDao.deleteRoleAndMenu(role.getId());
        roleDao.deleteRoleAndPermission(role.getId());
        //再添加角色与菜单、权限的关联关系
        this.setRoleRelevance(role);
    }

    /**
     * 根据ID删除角色
     * @param id    角色ID
     * @throws BusinessException    可能出现的业务异常
     */
    @Override
    public void delete(int id)  throws BusinessException{
        //首先检查角色与权限、菜单、用户的关联关系
        int count = roleDao.findCountByRoleId(id);
        if (count > 0 ){
            throw new BusinessException("删除失败，该角色与权限、菜单、或者用户有关联，请先解除关联关系！");
        }
        //不存在关联、可以删除
        roleDao.delete(id);
    }

    /**
     * 根据ID查询角色信息
     * @param id    角色ID
     * @return  角色实例
     */
    @Override
    public Role findById(int id) {
        Role role = roleDao.findById(id);
        //权限对象、用户对象、菜单对象
        Set<Integer> permissionIds = roleDao.findPermissionIdByRoleId(id);
        LinkedHashSet<Integer> menuIds = roleDao.findMenusIdByRoleId(id);
        role.setMenuIds(menuIds);
        role.setPermissionIds(permissionIds);
        return role;
    }

    /**
     * 分页查询角色信息
     * @param currentPage   当前页码
     * @param pageSize  每页条数
     * @param queryString   查询语句
     * @return  分页角色信息
     */
    @Override
    public PageResult findPage(Integer currentPage, Integer pageSize, String queryString) {
        //开启分页查询
        PageHelper.startPage(currentPage,pageSize);
        //调用dao层查询语句
        Page<Role> page = roleDao.findPage(queryString);
        List<Role> result = page.getResult();
        for (Role role : result) {
            Integer roleId = role.getId();
            //权限对象、用户对象、菜单对象
            Set<User> users = userDao.findUserByRoleId(roleId);
            Set<Permission> permissions = permissionDao.findPermissionsByRoleId(roleId);
            LinkedHashSet<Menu> menus = menuDao.findMenuByRoleId(roleId);
            role.setMenus(menus);
            role.setPermissions(permissions);
            role.setUsers(users);
        }
        //封装分页查询结果对象
        return new PageResult(page.getTotal(),page.getResult());
    }
}
